d353dd
@@ -16,12 +16,16 @@
 package org.wildfly.swarm.plugin.gradle;
 
 import java.io.File;
+import java.nio.file.Files;
+import java.nio.file.Path;
 import java.util.ArrayList;
 import java.util.List;
 import java.util.Properties;
 
 import groovy.lang.Closure;
 import groovy.util.ConfigObject;
+import org.gradle.api.plugins.JavaPluginConvention;
+import org.gradle.api.tasks.SourceSet;
 import org.gradle.api.tasks.bundling.Jar;
 import org.gradle.api.Project;
 
@@ -29,6 +33,7 @@
import org.gradle.api.Project;
  * @author Bob McWhirter
  */
 public class SwarmExtension {
+    private static final String MODULE_DIR_NAME = "modules";
 
     private Project project;
 
@@ -50,7 +55,19 @@
public class SwarmExtension {
 
     public SwarmExtension(Project project) {
         this.project = project;
-        this.moduleDirs.add(new File(this.project.getBuildDir(), "resources/main/modules"));
+        Path resourcesOutputDir = project.getConvention()
+                .getPlugin(JavaPluginConvention.class)
+                .getSourceSets()
+                .findByName(SourceSet.MAIN_SOURCE_SET_NAME)
+                .getOutput()
+                .getResourcesDir()
+                .toPath()
+                .resolve(MODULE_DIR_NAME);
+        if (Files.isDirectory(resourcesOutputDir)) {
+            File moduleDir = resourcesOutputDir.toFile();
+
+            this.moduleDirs.add(moduleDir);
+        }
     }
 
     public void properties(Closure<Properties> closure) {
